from .base import BaseHead
from .stgcn_head import STGCNHead

__all__ = [
    'BaseHead', 'STGCNHead'
]